package leetCode;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * 18. 四数之和
 * https://leetcode.cn/problems/4sum/description/
 */
class Solution2 {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        Arrays.sort(nums);//排序
        List<List<Integer>> ret = new ArrayList<>();
        int len = nums.length;
        for(int i = 0; i < len; ) {
            long targetA = target - nums[i];
            for(int j = i + 1; j < len; ) {
                int left = j + 1;
                int right = len - 1;
                long targetB = targetA - nums[j];//在j下标后面，要找的两数之和的目标值
                while(left < right) {
                    int sum = nums[left] + nums[right];
                    if(sum < targetB) {
                        left++;
                    } else if(sum > targetB) {
                        right--;
                    } else {
                        //找到两数之和等于targetB的数了，在链表中添加i，j，left，right下标位置的数组元素
                        ret.add(new ArrayList<>(Arrays.asList(nums[i],nums[j],nums[left],nums[right])));
                        //j下标位置后的left和right，还要继续往下找
                        left++;
                        right--;
                        //去重
                        while(left < right && nums[left] == nums[left - 1]) left++;
                        while(left < right && nums[right] ==nums[right + 1]) right--;
                    }
                }
                //给j下标的位置去重
                j++;
                while(j < len && nums[j] == nums[j - 1]) j++;
            }
            //给i下标的位置去重
            i++;
            while(i > 0 && i < len && nums[i] == nums[i - 1]) i++;
        }
        return ret;
    }
}